//
//  WBStatusPictureView.swift
//  SinaWeibo_Swift
//
//  Created by TianZhen on 16/4/15.
//  Copyright © 2016年 TianZhen. All rights reserved.
//

import UIKit
import SDWebImage

/// collectionViewcell的间距
private let HMStatusPictureViewCellMargin: CGFloat = 4

/// cell的最大列数为3列,列数使用整数
private let HMStatusPictureViewMaxColumn = 3

// 计算cell的宽和高
private let HMStatusPictureViewItemWH = (UIScreen.mainScreen().bounds.width - 2 * HMStatusCellMargin - (CGFloat(HMStatusPictureViewMaxColumn) - 1) * HMStatusPictureViewCellMargin) / CGFloat(HMStatusPictureViewMaxColumn)

class WBStatusPictureView: UICollectionView {

    private let ReuseIdentifier = "ReuseIdentifier"
    
    /// 配图NSURL数组
    var pictureURLs: [NSURL]? {
        didSet {
            // 当别人设置配图的url后,我们就来计算配图的高度
            let size = calcViewSize()
            
            // 自己来约束大小,当cell给pictureURLs之后计算出宽高,然后重行修改约束
            self.snp_updateConstraints { (make) -> Void in
                make.size.equalTo(size)
            }
            //            HMPrint("配图大小: \(size)")
            // 因为cell的复用重新刷新数据
            self.reloadData()
        }
    }
    
    /// 计算配图的宽高
    private func calcViewSize() -> CGSize {
        // 设置collectionView layout
        layout.itemSize = CGSize(width: HMStatusPictureViewItemWH, height: HMStatusPictureViewItemWH)
        layout.minimumInteritemSpacing = 0
        layout.minimumLineSpacing = 0
        
        // 判断图片的数量
        let count = pictureURLs?.count ?? 0
        
        // 没有图片
        if count == 0 {
            //            HMPrint("没有图片")
            return CGSizeZero
        }
        
        // 1.张图片
        if count == 1 {
            // 默认大小
            var size = CGSize(width: HMStatusPictureViewItemWH, height: HMStatusPictureViewItemWH)
            
            // 返回图片本来的大小,拿到图片来获取大小
            // key: 图片的URL地址
            let url = pictureURLs!.first!
            // 缓存图片有可能拿得到,也有可能那不到
            if let image = SDImageCache.sharedImageCache().imageFromDiskCacheForKey(url.absoluteString) {
                // 点坐标:像素 / 2, 在 SDWebimage 3.7.3以后
                size = image.size
                size = CGSize(width: size.width * UIScreen.mainScreen().scale, height: size.height * UIScreen.mainScreen().scale)
                //                HMPrint("单张图片size: \(size)")
            }
            
            // 如果图片太小,或太大,都处理一下
            if size.width < 40 {
                size.width = 40
            }
            
            let screenWidth = UIScreen.mainScreen().bounds.width
            
            if size.width > screenWidth * 0.5 {
                size.width = screenWidth * 0.5
            }
            
            layout.itemSize = size
            return size
        }
        
        layout.minimumInteritemSpacing = HMStatusPictureViewCellMargin
        layout.minimumLineSpacing = HMStatusPictureViewCellMargin
        // 多张图片,列数不确定
        // 2张和4图片显示2列,其他都显示3列
        var column = HMStatusPictureViewMaxColumn
        if count == 2 || count == 4 {
            column = 2
        }
        
        // 服务器开发用得比较多,分页,保证列数是整数
        // 2 + 2 - 1 / 2 = 1
        // 3 + 3 - 1 / 3 = 1
        // 4 + 2 - 1 / 2 = 2
        // 5 + 3 - 1 / 3 = 2
        // 6 + 3 - 1 / 3 = 2
        // 7 + 3 - 1 / 3 = 3
        // 行数 = (总个数 + 列数 - 1) / 列数
        let row = (count + column - 1) / column
        
        // 计算宽度和高度
        // 宽度 = 列数 * item宽度 + (列数 - 1) * 列间距
        let widht = CGFloat(column) * HMStatusPictureViewItemWH + (CGFloat(column) - 1) * HMStatusPictureViewCellMargin
        
        // 高度 = 行数 * item高度 + (行数 - 1) * 行间距
        let height = CGFloat(row) * HMStatusPictureViewItemWH + (CGFloat(row) - 1) * HMStatusPictureViewCellMargin
        return CGSize(width: widht, height: height)
    }
    
    private var layout = UICollectionViewFlowLayout()
    
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    init() {
        super.init(frame: CGRectZero, collectionViewLayout: layout)
        
        // 设置背景颜色
        backgroundColor = UIColor.clearColor()
        
        // 设置数据源
        self.dataSource = self
        
        // 注册cell
        self.registerClass(HMStatusPictureViewCell.self, forCellWithReuseIdentifier: ReuseIdentifier)
    }
}

// MARK: - 扩展实现数据源
extension WBStatusPictureView: UICollectionViewDataSource {
    /// 返回多少组,不写就是返回一组
    //    func numberOfSectionsInCollectionView(collectionView: UICollectionView) -> Int {
    //        <#code#>
    //    }
    
    /// 对应的section返回多少个cell
    func collectionView(collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return pictureURLs?.count ?? 0
    }
    
    /// 每个cell张什么样子
    func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
        // 去缓存池中获取,如果缓存池中没有就会使用注册的cell来创建cell,并返回
        let cell = collectionView.dequeueReusableCellWithReuseIdentifier(ReuseIdentifier, forIndexPath: indexPath) as! HMStatusPictureViewCell
        
        //        cell.backgroundColor = UIColor.randomColor()
        
        // 给cell设置URL
        cell.imageURL = pictureURLs![indexPath.item]
        
        return cell
    }
}

/// 自定义cell
class HMStatusPictureViewCell: UICollectionViewCell {
    
    // 每个cell需要一个图片的URL来取下载图片
    /// didSet是存储性数性才有的
    var imageURL: NSURL? {
        didSet {
            bkgImageView.sd_setImageWithURL(imageURL!)
        }
    }
    
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    
    override init(frame: CGRect) {
        super.init(frame: frame)
        
        prepareUI()
    }
    
    private func prepareUI() {
        bkgImageView.contentMode = UIViewContentMode.ScaleAspectFill
        bkgImageView.clipsToBounds = true
        
        // 1.添加控件
        contentView.addSubview(bkgImageView)
        
        // 2.添加约束
        bkgImageView.snp_makeConstraints { (make) -> Void in
            make.edges.equalTo(contentView)
        }
    }
    
    // MARK: - 懒加载
    /// 背景图片
    private lazy var bkgImageView: UIImageView = UIImageView()
}
